/*
** client.c -- a stream socket client demo
*/

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>

#include <arpa/inet.h>

#include "client.h"

#define PORT "3490" // the port client will be connecting to



// get sockaddr, IPv4 or IPv6:
void *get_in_addr(struct sockaddr *sa)
{
    if (sa->sa_family == AF_INET) {
        return &(((struct sockaddr_in*)sa)->sin_addr);
    }

    return &(((struct sockaddr_in6*)sa)->sin6_addr);
}

void ping(void)
{
	int i;
	for (i = 2; i <= 32; i++)
	{
		free(userLists[i]);
		userLists[i] = NULL;
		char ipAddress[15];
		sprintf(ipAddress, "142.104.72.%d", i);
		userLists[i] = pingSingle(ipAddress);
	}
}

userFileList_t* pingSingle(char* ip)
{
    userFileList_t *userFiles = malloc(sizeof(userFileList_t));
    int sockfd, numbytes;
    char buf[MAXDATASIZE];
    struct addrinfo hints, *servinfo, *p;
    int rv;
    char s[INET6_ADDRSTRLEN];

    memset(&hints, 0, sizeof hints);
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;

    if ((rv = getaddrinfo(ip, PORT, &hints, &servinfo)) != 0) {
        fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(rv));
        return NULL;
    }

    // loop through all the results and connect to the first we can
    for(p = servinfo; p != NULL; p = p->ai_next) {
        if ((sockfd = socket(p->ai_family, p->ai_socktype,
                p->ai_protocol)) == -1) {
			perror("client: socket");
            continue;
        }

        if (connect(sockfd, p->ai_addr, p->ai_addrlen) == -1) {
            close(sockfd);
            perror("client: connect");
            continue;
        }

        break;
    }

    if (p == NULL) {
        fprintf(stderr, "client: failed to connect\n");
        return NULL;
    }

    inet_ntop(p->ai_family, get_in_addr((struct sockaddr *)p->ai_addr),
            s, sizeof s);
    printf("client: connecting to %s\n", s);

    freeaddrinfo(servinfo); // all done with this structure

    if ((numbytes = recv(sockfd, buf, MAXDATASIZE-1, 0)) == -1) {
        perror("recv");
        return NULL;
    }

    buf[numbytes] = '\0';
	char fileBuf[44000];
	if (strcmp(buf, "623hp") == 0)
	{
		strcpy(userFiles->ipAddress, ip);
		if ((numbytes = recv(sockfd, &userFiles->fileNameList, sizeof(char)*25*30, 0)) == -1) {
			perror("recv");
			return NULL;
		}
		int currentFile = 0;
		while (strcmp(fileBuf, "END OF FILES") != 0)
		{
			if ((numbytes = recv(sockfd, fileBuf, 44000-1, 0)) == -1) {
        			perror("recv");
        			return NULL;
			}


			char* actualFile = malloc(sizeof(char) * (strlen(fileBuf)+1));
			strcpy(actualFile, fileBuf);
			userFiles->fileList[currentFile] = actualFile;
			currentFile++;
		}
	}
    printf("client: received '%s'\n",buf);

    close(sockfd);

    return userFiles;
}

foundFile_t* query(char *queryString)
{
	int i;
	for (i = 2; i < 33; i++)
	{
		if (userLists[i] != NULL)
		{
			int j;
			for (j = 0; j < MAX_CHILDREN; j++)
			{
				char *parsedCommand;
				char *params = NULL;
	
				params = strtok(userLists[i]->fileNameList[j], "/");
				int paramCounter;
				for(paramCounter = 0; params != NULL; paramCounter++)
				{
					parsedCommand = params;
					params = strtok(NULL, " ");
				}

				if (strcmp(queryString, parsedCommand) == 0)
				{
					foundFile_t *match = malloc(sizeof(foundFile_t));
					strcpy(match->fileName, parsedCommand);
					strcpy(match->fileData, userLists[i]->fileList[j]);
					return match;
				}
			}
		}
	}
	
	fprintf(stdout, "File not found");
	return NULL;
}
